"""
测试套件，用来整体管理测试用例
"""
import unittest
from test_01demo import TestDemo
import test_01demo
from BeautifulReport import BeautifulReport  # 美化报告

#
# # 创建测试套件
# suite = unittest.TestSuite()
#
# # 创建一个用例加载器
# loader = unittest.TestLoader()
#
# # 用例加载
# # 第一种，按类名
# # suite.addTest(loader.loadTestsFromTestCase(TestDemo))
# # suite.addTest(loader.loadTestsFromTestCase(TestDemo2))
#
# # 第二种，按模块加载
# # suite.addTest(loader.loadTestsFromModule(test_01demo))
#
# # 第三种，按路径加载 pattern 如果文件是test开头，默认就是按此加载，可以不写
# suite.addTest(loader.discover(r"D:\code\python\python_learn\11_单元测试\01_unittest", pattern="test_*.py"))
#
# print("套件中的用例数量：", suite.countTestCases())
#
# # 运行用例
# # 创建运行器， 使用unittest.textRunner()是运行不了的
# # runner = unittest.TextTestRunner(verbosity=2)
# with open("./result.txt", "w", encoding="UTF-8") as f:
#     runner = unittest.TextTestRunner(verbosity=2, stream=f)  # stream是将结果保存到文件
#     runner.run(suite)


# 一行代码搞定
suite = unittest.defaultTestLoader.discover(r"/11_单元测试/01_unittest")
# with open("./result.txt", "w", encoding="UTF-8") as f:
#     runner = unittest.TextTestRunner(verbosity=2, stream=f)  # stream是将结果保存到文件
#     runner.run(suite)

runner = BeautifulReport(suites=suite)
runner.report(description="测试demo报告")
